couple of small changes