trying to highlight python code