Modified Migration file