clean up answers for assignment 1