code improvements and one new figure is added