Added tests for yields and blocks