Added main class and removed test documents