Fix comments for public functions