Cleanup before starting assignment 2