Added set_code to mongo migration.