Homework part 2 first commit