Added a simple test class