improvements in figures