Untitled

def menu():
    print('1 - Enter population data ')
    print('2 - View the data')
    print('3 - Exit')
    return input('Choose any option: ')

while True:
    choice = menu()
    if choice == '1':
        get_data()
    elif choice == '2':
        display_data()
    elif choice == '3':
        break
    else:
        print("Invalid choice")