Made the program accept arguments instead of using prompt