Scale dataset better