Add some traits