Two changes - Removed references to test range and added imputation before separating out feature set