Boa tarde, Matheus! Tudo bem?
Ficou ótima sua solução para o desafio, aplicou corretamente a condição e está bem organizado o seu código, trazendo assim uma boa leitura.
No mais, recomendo que sempre compartilhe o seu código em forma de código para facilitar a visualização, já que, em outros contextos, com um código mais extenso, ficaria difícil analisar o código por uma imagem.
Enfim, bons estudos!