Spoiler - Here's my solution in 10 lines

Francisco_Assumpcao_code