Add type annotation