added a simple infection model