Wrote comments and small adjustments.