add comment
fix bug