Add tests for blocks