Begun work on SEIR extension