Add spec for taking a block