add main for test