Add type annoation