Add penalty recognition functionality (not 100% trustworthy)