Correct python sample in README for the function usage

Tests: