Fix comments for better understanding