Add architecture diagram back in