Added main test code