refactor code
Merge pull request #3 from viktoriastadub/lecture4

refactor code