Updated the UI to be a bit prettier/more sensible.