Finished week4 assignment