Add migration for modules