Added ability to load traits