Could you also add tests for blocks taking arguments? Thanks.