cleaning up code to leave lab with