Added a test class for testing purposes.