Add a callback method for VB in order to save custom stats during learning