included new equation to output