Give onLoad a callback

Similar to the way onSave works now