diff --git a/bin/pwnagotchi b/bin/pwnagotchi
index 302b9131..787ce174 100755
--- a/bin/pwnagotchi
+++ b/bin/pwnagotchi
@@ -176,8 +176,12 @@ def pwnagotchi_cli():
                     os.system("rm /root/.auto-update && systemctl restart pwnagotchi")
                 os.system("systemctl restart pwnagotchi")
                 print("Okay, give me a couple minutes. Just watch pwnlog while you wait.")
-            print("Okay, guess not!")
-        print("You are currently on the latest release, v%s." % pwnagotchi.__version__)
+            elif user_input.lower() in ('n', 'no'):
+                print("Okay, guess not!")
+            else:
+                print("Invalid input.")
+        else:
+            print("You are currently on the latest release, v%s." % pwnagotchi.__version__)
         sys.exit(0)
 
     config = utils.load_config(args)
diff --git a/builder/data/usr/bin/pwnlib b/builder/data/usr/bin/pwnlib
index 8cd1670c..d6780b7d 100755
--- a/builder/data/usr/bin/pwnlib
+++ b/builder/data/usr/bin/pwnlib
@@ -39,10 +39,8 @@ reload_brcm() {
 start_monitor_interface() {
   rfkill unblock all
   ifconfig wlan0 up
-  sleep 3
   iw dev wlan0 set power_save off
   iw phy "$(iw phy | head -1 | cut -d" " -f2)" interface add wlan0mon type monitor
-  sleep 2
   rfkill unblock all
   ifconfig wlan0 down
   ifconfig wlan0mon up
diff --git a/pwnagotchi/ui/hw/libs/waveshare/oledhat/epd.py b/pwnagotchi/ui/hw/libs/waveshare/oledhat/epd.py
index 7c1be016..ce5a690e 100644
--- a/pwnagotchi/ui/hw/libs/waveshare/oledhat/epd.py
+++ b/pwnagotchi/ui/hw/libs/waveshare/oledhat/epd.py
@@ -2,11 +2,12 @@ from . import SH1106
 from . import config
 
 # Display resolution
-EPD_WIDTH       = 64
-EPD_HEIGHT      = 128
+EPD_WIDTH = 64
+EPD_HEIGHT = 128
 
 disp = SH1106.SH1106()
 
+
 class EPD(object):
 
     def __init__(self):