Added UI improvements and level tweaks