Optimising your code with math