Add logging to migration