@wip Add documentation for traits