Fix auto-comment for function with return type

Also add some tests for label-be on function and task