Fixed small issues founded after merging
Merge pull request #101 from Ruszrok/fix-for-kirill-lab3

Fixed small issues founded after merging.