Adding code for python sample v1