Correct some logic. Add spec