Updates to flask app