Assignment 2

Completed code