Small fixes for the theme