Half done example of using the State monad StateM
Clean up test
Removed temp variable