Added comments in code