need to finish model