Eso es bueno, pero aquí lo importante es que hayas entendido lo que cada linea del código hace, con ello tendrás más probabilidades de hacer modificaciones más complejas en el futuro, cuando te topes con un código que te sirva, independientemente de si funciona bien a la primera, analízalo y compréndelo.
Bien por solucionar el problema