Can you write a rust program to do the math?