Use implicit return in load
Benchmark loading as well