Diego Costa quickly finds the net in first appearance since leaving Chelsea