Added main method for test