Add more arguments to customize the figure