Added a loop to keep program running.