Improves python support