def main():

    # Initial With Device Handshake
    serial_socket = hs.choose_serial_ports()

    # Welcome message
    print("\n\nWelcome to the rocket program!")

    # Print Menu
    print_menu()
    
    while True:
       
        # Get mode from user
        mode = raw_input("> ")

        # Mode 'c': Collect Data
        if (mode == 'c'):
            print("\tEntering collect data mode...")
            collect_data(serial_socket)
        # Mode 'd': Download Data
        elif (mode == 'd'):
            print("\tEntering download data mode...")
            download_data(serial_socket)
        else:
            print("\tIncorrect entry. Try again!")
            time.sleep(0.5)
            print_menu(0.1)
def main():



    # Handshake
    serial_socket = hs.choose_serial_ports()

    print("Welcome to the rocket program menu!")
    print("\n")
    print("      Menu      ")
    print("----------------")
    print("[c]ollect  data ")
    print("[d]ownload data ")
    print("----------------")
    print("\n")

    mode = raw_input("> ")

    if (mode == 'c'):
        print("Entering collect data mode...")
        collect_data(serial_socket)

    elif (mode == 'd'):
        print("Entering download data mode...")
        download_data(serial_socket)