Improve generated plots