Minor changes required to get the module pushed to npm