finished Assignment 1