changed main to accept one argument