Add more code to application.py