modified the model and added more plots
First commit of the full wheat imputation