Implicit state management (#103)

* Rebase and fix merge conflict

* formatting

* MR comments

* clang format

* Fix double StateNew

---------

Co-authored-by: Iman Tabrizian <[email protected]>

Author: jamied157