second commit, added some figures, working on the imputation