homework_4 fixed I hope