Még egy kis takarítás + refactoring