Assignment 2 for 402