Added some traits for models