Updated test code for PID module