Pode usar este código atualizado. Se precisar fazer mais alguma alteração ou tiver alguma dúvida, é só me dizer!