dodata funkcionalnost izbora klasa