Support lambda-argument instead of block

Fix #7