Minor style fixup in ywprim.py

Simplifying some issues.