diff --git a/init.lua b/init.lua
index 73e1011..fd2c1c6 100644
--- a/init.lua
+++ b/init.lua
@@ -22,6 +22,7 @@ require('paq')({
 	'hrsh7th/cmp-nvim-lua',
 	'hrsh7th/vim-vsnip',
 	'hrsh7th/cmp-vsnip',
+	'wakatime/vim-wakatime',
 })
 require('colorscheme')
 require('config')
diff --git a/lua/plugins/lines.lua b/lua/plugins/lines.lua
index 651df01..e16dc9e 100644
--- a/lua/plugins/lines.lua
+++ b/lua/plugins/lines.lua
@@ -45,14 +45,21 @@ require('lualine').setup({
 -- Credit to the original author of cokeline for creating most of this config
 require('cokeline').setup({
 	default_hl = {
-		focused = {
+		--[[focused = {
 			fg = get_hex('Normal', 'fg'),
 			bg = get_hex('Background', 'bg'),
 		},
 		unfocused = {
 			fg = get_hex('Comment', 'fg'),
 			bg = get_hex('Background', 'bg'),
-		},
+		},--]]
+		fg = function (buffer)
+			if buffer.is_focused then
+				return get_hex('Normal', 'fg')
+			end
+			return get_hex('Comment', 'fg')
+		end,
+		bg = function(_) return get_hex('Background', 'bg') end
 	},
 
 	components = {