moved yield inside of .container