fix style and add comment