Add a comment and fix the code style