Small changes to figure scripts