clean up code using clojure.walk