Slight refactoring
Added a lookup function (mirrros Map.lookup)