couple of little changes