Bug fixes, style updates