Made things look pretty and added a save image and stuff.