Fixing python support a bit