Added more comments, slight change