Updates to Flask app files to get it working