Moved TreeCapacator integration to proper location

This closes #297.