modify for easy figure inclusion