Updates

Bug fixes