Unit 2, Lesson 2, Part 5 - Milestone 5

Fix up script to return proper discount regardless of type