Modificada línea

 No veo que haya ningún fallo, pero creo que te ocupa demasiado y se puede conseguir que el programa haga lo mismo con menos código. Solo he cambiado una línea para que te quede más estético.