Little modification

Start to understand the source code