Save button + Some cleanup

Signed-off-by: Xmister <xmisterhu@gmail.com>