Add callbacks for render method