Add python maths stuff