Tube model almost working