Adding diagram div back in