Add callbacks for dis shiz