Moved node state to infection object