assignment 3: add calculations