Added Virus Model Properly. Cleaned up project.