Updates

Minor changes. Updated comments