Moved neural networks into threads, optimised dataset class(s)