First (unfinished) stab at python wrapper