Added class stubs from teacher (task 2)