Add FLASK support