Use example lexer from book. Add test case for if statement