Add counting Python program