Rough code created for flask