diff --git a/01 - JavaScript Drum Kit/index-START.html b/01 - JavaScript Drum Kit/index-START.html
index 4070d32767..b69398c5b3 100644
--- a/01 - JavaScript Drum Kit/index-START.html
+++ b/01 - JavaScript Drum Kit/index-START.html
@@ -58,9 +58,37 @@
+ if (!audio) return; // stop the function from running all together
+
+ audio.currentTime = 0; // rewind to the start
+ audio.play();
+ key.classList.add('playing');
+ }
+
+ function removeTransition (e) {
+ if (e.propertyName !== 'transform') return; // Skip it if it's not a transform
+
+ this.classList.remove('playing');
+ }
+
+ window.addEventListener('keydown', playSound);
+ keys.forEach(key => key.addEventListener('transitionend', removeTransition));
+ // window.addEventListener('keyup', function __keyListener__ (e) {
+ // const key = document.querySelector(`.key[data-key="${e.keyCode}"]`);
+ // if (!key) return; // stop the function from running all together
+
+ // key.classList.remove('playing');
+ // });
+
+
+