Highlight transmembrane segments in Pymol