Added comments into code