added improved plot