refactoring i poprawienie funkcjonalnosci funkcji Model.java