added main class for testing