Added logic for saving files.
Added a little styling.