Add python iterator