Added test class with some tests