Add comments and fix some brackets
Fix comment