Step 2 of the exercise, changed random # generator function