Fixes #4 Add some spec