Added if __name__ == __main__ code
Fixed menu options not being selectable because values were integers instead of strings