Significantly improve flask assistant