Post hw10 clean up