Bugs

Bug fixes