Second update, added Main test class