Added line graph, working on imputation