Assignment 3 Part 4 - Cleanup