UI updates, fixes #34