change load function for a new architecture