added small test class