Added visuals to easier understand classes