minor improvements in plot helper