ML model working again
mergining ML stuff back in