Add PID, fix bug