model almost working