Add comments
Code clean