Add type annotations