old_contents: #!/usr/bin/python
  
  import onepk_helper
  import yaml
  import sys
  
  def main(args):
      ''' main '''
      rtrs = yaml.load(args[0])
      for rtr in rtrs:
          ndev = NetworkDevice(**rtr)
          ndev.connect()
  
          print "Connection established to: %s" % ndev.net_element.properties.sys_name
          print ndev.net_element.properties.product_id
          print ndev.net_element.properties.SerialNo
  
  if __name__ == '__main__':
      try:
          sys.exit(main(sys.argv[1:]))
      except KeyboardInterrupt, e:
          print >> sys.stderr, "Exiting on user request.\n"
          sys.exit(1)
  
new_contents: #!/usr/bin/python
  
  from onepk_helper import NetworkDevice
  import yaml
  import sys
  
  def main(args):
      ''' main '''
      rtrs = yaml.load(args[0])
      for rtr in rtrs:
          ndev = NetworkDevice(**rtr)
          ndev.connect()
  
          print "Connection established to: %s" % ndev.net_element.properties.sys_name
          print ndev.net_element.properties.product_id
          print ndev.net_element.properties.SerialNo
  
  if __name__ == '__main__':
      try:
          sys.exit(main(sys.argv[1:]))
      except KeyboardInterrupt, e:
          print >> sys.stderr, "Exiting on user request.\n"
          sys.exit(1)
  
subject: fix import