Attempt at Assignment 2