Updated code logic and added some migration steps