Turns out sometimes writing more lines of python is better than writing fewer.