Added class-diagram