cleaned code
clarified comment