Improve python sytax highlight