kakarotto, si no he entendido mal, lo que tu quieres es acortar el script y hacerlo de forma más sencilla.
Mi propuesta es esta. No la he probado ni la he compilado porque estoy desde el móvil, pero en caso de tener errores, no creo que sean difíciles de detectar.
Si no es esto lo que quieres...