Simplifying math example to remove module