Better Python support