Migration Model 2

Bugs fixed now it runs properly