Updated the module exports because I am a bit of a numpty