# Problem getting indices to sync up

I'm not sure what is wrong with the following code:

```
# given a prime p, return all A_n representations of dimension = p^2
def rankrep(p):
bound = p**2
A = lambda i: WeylCharacterRing("A{0}".format(i), style = "coroots")
deg = []
for i in xrange(bound):
deg.append([])
fw = A(i+1).fundamental_weights()
temp = A(i+1)
for j in range(len(fw)):
deg[i].append(temp(fw[j]).degree())
return deg
```

But when I run it, it is unable to return anything, and tells me `Keyval Error 0`

. Should I not be using `append`

here?

I eventually hope to insert some `if`

trees testing whether a given `temp(fw[j].degree() == p**2`

, but I'm not sure how to pull those indices either. Any help would be greatly appreciated. Regards.

I've added this line of code ` [(i,j) for (i,deg_i) in enumerate(deg) for (j,deg_ij) in enumerate(deg_i) if deg_ij == p**2]` to pull the indices