Creating pymol color_by script