Restructure chat module