Clean up code from yesterday