Updated workflow to use the new modules