Forgot to import random module