Oi, Alex! Como vai?
Gostei da forma como você organizou o código e separou cada etapa com comentários. Isso ajuda bastante na leitura e manutenção do programa. Também foi uma boa escolha utilizar if, elif e else para classificar o IMC de maneira clara.
Dica: você pode testar outros valores de peso e altura para validar todas as condições do programa e conferir se cada mensagem aparece corretamente.