Minor Updates

Update code