Update-- Bug fixes