Create Array With Range Python Example Problem