issue #38, corrected uml
issu #38 , corrected code