finished Assignment 4