Implemented the "Imputing missing values" part of the assignment and minor changes.