Add lexer tests + fix direction