Make load.py more idiomatic.