WIP (NW): Everything is under State Monad

* Most of the tests are not working