Adding tests
Creating class