Update main

implementing a switch statement into program to make choice easier and changed loop in program to repeat.