Added Migration
Small changes