Improve code generation for maths ops
Maths function enhancements