address PR comments

reformat a comment block and re-import math package