Bug fixes, layout updates