Making it all-Flask