Added main class and test files