make model more flexible and add earthquake