ML HW2 bug fixed.