Add save button
Better looking