adding features to plotting