fix colors in python