Add callbacks for simulator