Finished assignment for Unit 1