Almost finished medical_test.py