tesztesetek + némi javítás a tesztek miatt