Changed to class relationship diagram and updated examples for v0.3.1 fix