using dic in  python