ztree: Wrap in setup function

evil-ztree.el

@@ -24,12 +24,13 @@
 ;; see <http://www.gnu.org/licenses/>.
 
 ;;; Commentary:
+;; Evil bindings for ztree.
 
 ;;; Code:
-
 (require 'evil)
 (require 'ztree)
 
+(defun evil-ztree-setup ()
   (evil-set-initial-state 'ztree-mode 'motion)
 
   (evil-define-key 'motion ztree-mode-map
@@ -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