added change so that you can specify saving trait graphs