Add some additional class traits