Futher work on implementing code migrations