Now, I've had better understanding of the algorithm, made the code more readable.

Signed-off-by: Thien Nguyen <nguyentm@ornl.gov>