Set up for flask testing