Finished Assignment

Finished code for Assignment 2.