refactor
Add specs