Add a helper function for implicit return