Implemented get_states and get_transitions