Assort Health Interview Question

Refactor some python code, open ended