Added comments and made final updates before submission