Change the framework to Flask