Added comment
Code cleanup