Some visual improvments.
Oops forgot the save