addressed comments on pr