Adding more docs to the state module