UI fixes #2