Fixed styling for flask