Fixed small UI bugs

fixes #238 (github)