Added docstrings
PATCH: Added one extra test and fixed a bug with not being to do a transition through the initial state