E questo è il codice che ti funziona, quindi è poco utile per dirti cosa non va.
Devi postare il codice modificato che NON fa quello che vorresti, in modo da poterti dire cosa sbagli.