Added simple unit tests for the "lookup" function