this should be `else if` or just `if` since you're `return`ing above