Updated the chat ui
Fixed a few bugs