Attempted to fix the imputation technique. I found and fixed one bug, but others persist