minor fix and add some comments