Python - Modificaciones a un programa

letters = [ '\n', 'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j', 'k', 'l', 'm', 'n', 'ñ', 'o', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y','z', ' ']

translate = [ '\n','aa', 'ab', 'ac', 'ad', 'ae', 'ba', 'bb', 'bc', 'bd', 'be', 'ca', 'cb', 'cc', 'cd', 'ce', 'da', 'db', 'dc', 'dd', 'id', 'de', 'ea', 'eb', 'ec', 'ie', 'ed', 'ee', 'ii']

def codificar(mensaje):

	mensaje = mensaje.lower()

	codificado = ""

	for i in mensaje:

		codificado += translate[letters.index(i)]

	return codificado

def decodificar(mensaje):

	codificado = ""

	contador = 0

	for i in range(0, (len(mensaje) /2)):

		j = mensaje[contador]+mensaje[contador+1]

		codificado += letters[translate.index(j)]

		contador+=2

	return codificado

dec = 0

while(dec != 3):

        dec = int(raw_input("Digite 1. Para codificar o 2. para decodificar 3. para salir"))

        if(dec == 1):

                mensaje = raw_input("Digite el mensaje para codificar: ")

                print (codificar(mensaje))

        elif(dec == 2):

                mensaje = ("Digite el mensaje para decodificar: ")

                print (decodificar(mensaje))

        else:

                dec = 3