Issue 21 add unit specs