Add option to run program infinitely