Cleans up code for Lab 2A submission.