Updated comments indicating the math better