ENH: Adding test