Update socode.py

Added my addition