Finished question 1 of the first homework