Ошибка в программе Python 3.2

# Дана квадратная матрица A порядка M.
# Найти сумму элементов ее главной диагонали, т. е. диагонали, содержащей следующие элементы:
# A1,1, A2,2, A3,3, …, AM,M.

a=0
z=[]

m = int(input ('M= '))

for i in range(m):
for j in range(m):
z[j]=1
if i==j:
a=a+z[j]
print ('S= ',a)

В консоле Error:

z[j]=1
IndexError: list index out of range