More pythonic source