m = session.models[0]
residues = m.residues
for r in m.residues:
    if r.principal_atom is not None:
        r.worm_radius = 0.02*r.principal_atom.bfactor + 0.1