A couple small changes