Homework #2 finished