Adding code to allow math parameter