Combined two migration scripts into one