- Clean up code