Python Random data generation Exercise