improve pry integration