code to cleanup, impute the input data and also to predict using simple models