Migration draft added and models updated