Minor changes to virus module to include UDP virusses