some UI fixes :|