Added main class for testing