almost done with model probably