update docblock

This updates all docblocks to with the right type annotations.