Add function to add new states
Add some basic tests
Add some basic tests