Make it into a flask app