Add return type