Add my original color scheme and improve python syntax