Updated module and docs