Improving state docs

Ready to be merged @thatch45 !