Add getBeginStates method structure