updated Homework 4