UI fixes, new save button