edit main for test