Clean up code for turning in project