Added comments + minor updates.