Add code to run calculations