Add homework 3, part 2