ENH: small changes