Optimize and simplify load method