finished Assignment 2