Implementing same solution in clojure
Using update-in to increment ways