adding a test class
updated test code