add comments and clean up
add comments and clean up
I had to review my code cause it corrupted by github