Added UI elements for mode selection