Finished week2 assignment