Lots of UI stuff
Fixing chat ui