Added comments. Small touchups