Update lexer.go

Just simplify some code for lexer