minor changes to get figure