ENH: Add testing