Adding Martha's traitgram function