add return to load function