Wrote out unit tests and added in the context as an additional parameter
passed to the lambda helper functions.