Update python prompt example