modify load function