updated module to work with the latest code