Add code to make specs pass