Attempt at Assignment 2

Not sure about this.