Add tests
Adding tests