Assigment 2 complete