Adding a restart method to States and call it