Working more on dataseting, meh