Added Comments

Added comments after code review.