Added epidemic and outbreak functions