Update lex.py

adjustment