Add some specs. 100% test coverage :)