finished assignment 2, added comments