Fix comments for functions with missing @return tag