Added options in main loop for what to do