Updated comments

Added comments in functions