Updated homework assginment