Minor plotting changes