committing adding the draft code for  if __name__ == '__main__': block