Add type hint for collection