Improve return values