Type annotation updates