More chat function and UI improvement