Moved migration related code to specific module
Moved migration related code to specific module