Add return annotation