Improved migrations setup