Add return statement in estimate()