Add more unit tests for lookup table.
Add backprop.