Updated setup file with new module