Adjust kp-count script to new format of kp