add pretty plot plotting