small changes to module