Adding volume control and other UI tweaks