Added the SetModule function.