Minor edits on plotting.