close to being finished with assignment 2