加註解

順便簡單重整了一下原始碼