diff --git a/micropython/examples/pico_display/demo.py b/micropython/examples/pico_display/demo.py
index 48fd7698..fdc90ec5 100644
--- a/micropython/examples/pico_display/demo.py
+++ b/micropython/examples/pico_display/demo.py
@@ -1,16 +1,15 @@
 import time, random
 import picodisplay as display 
 
-WIDTH = 240
-HEIGHT = 135
-
-display_buffer = bytearray(WIDTH * HEIGHT)            
-display.init(display_buffer)
-display.set_backlight(1.0)
-i = 0
 width = display.get_width()
 height = display.get_height()
 
+display_buffer = bytearray(width * height * 2)  # 2-bytes per pixel (RGB565)
+display.init(display_buffer)
+
+display.set_backlight(1.0)
+
+
 class Ball:
     def __init__(self, x, y, r, dx, dy, pen):
         self.x = x