Lab01 clean up