fixed up UI to make things easier to understand