More cleaning up for data class, cleaning up main as well
changed data class to RawData class and created MLData class instead of using structs for training and test data