Finished Week5 assignment