Move data to diff class; makes balancing easy