Bug fixes, update