Unnessecary if check was removed
Classes diagramm was added