Updated widget to use Math helper