improve plotting