working on virus_predictor challenge