Add type annotations to producer