assignment 4 clean up