small plotting changes