Fix diff colors, add python stuff