Finished week2 problem