Redesigned the program a little.