add dimension parameter in load