small change

small change
update with priority_queue

update with priority_queue