Small fixes

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)

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

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):