Better flask integration