Added spec & got it to pass