diff --git a/demo.gif b/demo.gif
index 7f8cd29..7aa0a0f 100644
Binary files a/demo.gif and b/demo.gif differ
diff --git a/ftdetect/aichat.vim b/ftdetect/aichat.vim
new file mode 100644
index 0000000..f953b68
--- /dev/null
+++ b/ftdetect/aichat.vim
@@ -0,0 +1 @@
+autocmd BufNewFile,BufRead *.aichat setfiletype aichat
diff --git a/plugin/vim-ai.vim b/plugin/vim-ai.vim
index 6f14f76..1e9407f 100644
--- a/plugin/vim-ai.vim
+++ b/plugin/vim-ai.vim
@@ -8,6 +8,7 @@ function! ScratchWindow()
   setlocal buftype=nofile
   setlocal bufhidden=hide
   setlocal noswapfile
+  setlocal ft=aichat
 endfunction
 
 function! AIRun(...) range
diff --git a/syntax/aichat.vim b/syntax/aichat.vim
new file mode 100644
index 0000000..ecd8459
--- /dev/null
+++ b/syntax/aichat.vim
@@ -0,0 +1,5 @@
+syntax match aichatRole ">>> system"
+syntax match aichatRole ">>> user"
+syntax match aichatRole "<<< assistant"
+
+highlight default link aichatRole Comment