Allowing union types in the return type