small changes in zad 3