Add support for flask