Fixed state migration
Add drop down menu to change states on save