Add states for state mechine and made them pass.