update docblock

This updates all docblocks with the right type annotations.