Change condition for changing discount
Clean up test.py