Implement lens-compatible `at` function