Add homework 3 code.