Add optional code word checking