Small edits to kazr plotting functions