improving dataset