Add in visualization of blocks