Minor improvements for the flask theme