[PyMOL] Question about python commands in PyMOL

[Hrutvik B. <hru...@gm...>]

Hello,

I'm writing a python script that loads in PDBs then selects certain chains
colors them according to a standardized chart. However, I've run into the
issue where my when I try running my script in Pymol via the run command it
just displays the text on the python script but doesn't actually execute
the command. I've pasted a sample script below. Please let me if there's a
way to fix this. Thank you for your help.
Sincerely, Hrutvik.

from pymol import cmd, stored

def load( arg1):

    print ("load 1u8q.pdb")
    print ("color red, (chain y)" + "\n")

    return (arg1)

cmd.extend( "load", load );
cmd.extend( "color", color );