yield problem sorted