Refactor and/or remove functions in file state.py.
state.py
Also add unit tests and more documentation.