almost done with assignment
first part is done