Can use ternary operator here:

'''python
return 'DONE' if status == 'TODO' else 'TODO'
'''