Changed rake task to support modules.