增加一些注释，简化 if-else 语句