Помогите, пожалуйста! Знаю, что задача простая, но второй день не могу ее одолеть

Есть вот такой код:
import re import os a = ['123', '456', '789'] b = input("Введите первый элемент числа: ") b1 = b.upper() os.system('cls') for str in a: if re.match(b1, str): print (str)
А мне нужно чтобы получилось вот что вроде этого:
import re import os a = ['123', '456', '789'] while True: os.system('cls') b = input("Введите первый элемент числа, или 999 для выхода: ") b1 = b.upper() if b1 == "999": break elif b1 != "999": for str in a: if re.match(b1, str): print (str)