homework_4 fixed