diff --git a/detect_buzz.py b/detect_buzz.py
index 0daa9af..dd71749 100644
--- a/detect_buzz.py
+++ b/detect_buzz.py
@@ -1,8 +1,11 @@
 #!/usr/bin/env python3
 # -*- coding: utf-8 -*-
-# See https://gpiozero.readthedocs.io/en/stable/api_input.html
+"""
+Using GPIO to detect button
+See https://gpiozero.readthedocs.io/en/stable/api_input.html
+"""
 from gpiozero import Button
 buzzer = Button(16)
 
-button.wait_for_press()
+buzzer.wait_for_press()
 print("The button was pressed!")