Clarification

Makes some things a bit more Pythonic