Updated task return type