small change in load function