It's a very nice system, and I personally have an idea about it.
What do you think about modifying it to allow stealing only specific items so that it doesn't harm the players? For example, what do you think about making it steal only valuable upgrade tools, and not equipment like armor or...