Implementation of Vigenère cipher in Python