Add python wrapper and example