Adding some tests
Adding some tests