Update lex.py