* Added a feature to migration script