refactored code for screening traits with valid predictions..