fourth_week_python_assignment working code