Simplify python interface