ENH: Added first unit test