Tweak the module functionality