Add homework 3 exercise 1