Bug fixes, style fixes