minor fix

* arrange alignment
* update comment
* clean up