Working on getting traits function working