Adding more flask code