Added stuff to models