final day at internship, cleaning up code