WIP: working on fixing log linear model