working on a main dataset