Hacked in changes for how I want to use the module.