diff --git a/lua/cosmic/lsp/mappings.lua b/lua/cosmic/lsp/mappings.lua
index a07adf7..b6daee1 100644
--- a/lua/cosmic/lsp/mappings.lua
+++ b/lua/cosmic/lsp/mappings.lua
@@ -16,8 +16,8 @@ map('n', ']g', '<cmd>lua vim.diagnostic.goto_next()<cr>')
 map('n', 'ge', '<cmd>lua vim.diagnostic.open_float(nil, { scope = "line", })<cr>')
 map('n', '<leader>ge', '<cmd>lua require("telescope.builtin").lsp_document_diagnostics()<cr>')
 map('n', 'K', '<cmd>lua vim.lsp.buf.hover()<cr>')
-map('n', '<leader>ga', '<cmd>lua require("telescope.builtin").lsp_code_actions()<cr>')
-map('v', '<leader>ga', '<cmd>lua require("telescope.builtin").lsp_range_code_actions()<cr>')
+map('n', '<leader>ga', '<cmd>lua require("cosmic-ui").code_actions()<cr>')
+map('v', '<leader>ga', '<cmd>lua require("cosmic-ui").range_code_actions()<cr>')
 map('n', '<leader>gf', '<cmd>lua vim.lsp.buf.formatting()<cr>')
 map('v', '<leader>gf', '<cmd>lua vim.lsp.buf.range_formatting()<cr>')
 map('n', '<C-K>', '<cmd>lua require("lsp_signature").signature()<cr>')
diff --git a/readme.md b/readme.md
index 8b520d6..7a68281 100644
--- a/readme.md
+++ b/readme.md
@@ -154,12 +154,16 @@ Useful for non-plugin related changes such as editor settings or mappings.
 
 ### Rename
 
-<img width="859" alt="rename" src="https://user-images.githubusercontent.com/3721204/140592783-c9875634-805e-45f4-8547-6dcb5b5c7404.png">
+<img width="498" alt="Screen Shot 2021-12-10 at 4 22 28 PM" src="https://user-images.githubusercontent.com/3721204/145656501-e1aec4be-c8bc-4e59-8c2f-2d99d50bbea2.png">
 
 ### Rename change notification
 
 <img width="1792" alt="rename notification" src="https://user-images.githubusercontent.com/3721204/140592788-27c0f02a-e31b-4af7-8f0c-a49a988e295e.png">
 
+### Code Actions
+
+<img width="613" alt="Screen Shot 2021-12-10 at 3 37 38 PM" src="https://user-images.githubusercontent.com/3721204/145654798-84c88a69-414e-457b-b595-e76b767ea5d3.png">
+
 ### Hover
 
 <img width="1792" alt="hover" src="https://user-images.githubusercontent.com/3721204/140592798-b9dbbdee-8496-4998-a874-110a051a7672.png">