ztree: Wrap in setup function

evil-ztree.el

@@ -24,15 +24,16 @@
 ;; see <http://www.gnu.org/licenses/>.
 
 ;;; Commentary:
+;; Evil bindings for ztree.
 
 ;;; Code:
-
 (require 'evil)
 (require 'ztree)
 
-(evil-set-initial-state 'ztree-mode 'motion)
+(defun evil-ztree-setup ()
+  (evil-set-initial-state 'ztree-mode 'motion)
 
-(evil-define-key 'motion ztree-mode-map
+  (evil-define-key 'motion ztree-mode-map
     (kbd "<tab>") 'ztree-jump-side
     (kbd "<return>") 'ztree-perform-action
     (kbd "SPC") 'ztree-perform-soft-action
@@ -47,7 +48,7 @@
     "ZQ" 'quit-window
     "ZZ" 'quit-window)
 
-(evil-define-minor-mode-key 'motion 'ztreediff-mode
+  (evil-define-minor-mode-key 'motion 'ztreediff-mode
     "C" 'ztree-diff-copy
     "D" 'ztree-diff-delete-file
     "zH" 'ztree-diff-toggle-show-filtered-files
@@ -57,7 +58,7 @@
 
     ;; update
     "gr" 'ztree-diff-partial-rescan
-  "gR" 'ztree-diff-full-rescan)
+    "gR" 'ztree-diff-full-rescan))
 
-(provide 'init-evil-ztree)
+(provide 'evil-ztree)
 ;;; evil-ztree.el ends here