handled missing values in train data
Added error handling in Test.hs