Add type-hints