added solution for virus predictor