wrote a test class and it works