Fix ui a bit more
Merge pull request #30 from rmtmckenzie/Morgan

Fix ui a bit more