Adding an error for missing keys, ft. other minor fixes

* Added test coverage for batching batched functions.
* added an error if none of the keys `batch` is called with are present upon calling the batched function.
* Cleaned up `find_in_stack`.