Move States model to the store gem