Update type annotations