I think that after a return statement it's nicer/clearer to have `if` instead of `elif`