Start improving the UI