Clean up app.py