change to more flexible mean model