change hw4_solution1.py