Refactor FSM states to return their status, instead of having it being a member field (less error prone and helps states be AI architectures agnostic)