Improved GUI + Option to save code.