solved the shrinking dataset problem!