addressed comments from pr