Updated the style and added a save button