Added while loop to keep program running