modified prior; fixed linkage to test data in train