Mi sembra corretto ... quali dubbi hai?

(a parte il return che manca nel main)