Added main class file with basic test at the end