A couple of small changes