Cleanup and test math module