Erst mal zur Warteschleife:
ZH/ZL sind die Register 30 und 31 (weiss grade nicht in welcher reihenfolge). Das ist aber wohl das kleinste Problem. Der Befehl SBIW ist nicht bei allen AVR Controllern implementiert. Das ist ein spezieller Befehl für eine 16 Bit subtraktion: von der 16 Bit Zahl in ZH/ZL wird das in dem Beispiel jeweils 1 abgezogen.
Das BRNE ist ein normaler bedingter Sprung: Springe wenn Z-Flag nicht gesetz ist, also das Ergebnis der LEtzten Rechnung nicht 0 war.

Der Code ist umständlich könnte aber sonst fast funktionieren. Es fehlt noch die Initialliserung des Stackpointers.

Ein Erweitern des Probgramms ist sicher möglich. Man müßte da irgendwo die Tasten abfragen und dann ggf. in einen anderen Programmteil springen.