We typically like to avoid using `return` statements. Could you change this to use a full if/else?