Hice algunos comentarios

El código debe reconocer cualquier entrada y blindarse lo más posible
ante posibles errores del usuario