Fixed import (from numpy to random)