Added save feature, rewrote a few lines.