Add callbacks for state enter/exit and transition execution