Add tests for flask site