added code from lection 4
removed not needed classes