address comments from first review