Finished Homework part 4