Enhance UI

This makes the UI much prettier, including some basic controls.