hw3 clean up