Add customization for Pry