Removed class="diagram" that wasn't there in the sample code