Added comments

Little cleanup and added comments.