Kicsit átalakítottam a GUI-hoz