moved the tests to state
Merge pull request #6 from mattyw/upgrade-move-to-state

moved the tests to state