Added a test class and added some methods