improve plot