Added models & better stun