Added initial version of code that imputes missing values