small python changes