Added callback functionality to migrations