Converting list function to generator using yield