How To Return Something To A Menu In Python