almost done with assignment 7