made djustments
updated reference
removed yield