edited models to allow for less errors