Add deletion of State
Migrated the State test to a normal Unit test