Add function for loading set