Improve arithmetic grammar example