added in serial module
added in train and prediction capabilities