updated Homework 5