Added comments and small changes.