Some edits on the models