Extend signature annotations of closures

Signed-off-by: Niklas Koep <niklas.koep@gmail.com>