It looks like it can be more compact and still use the same concept, but I'm too lazy to figure that out right now. If you make it smaller, be sure to provide a save id and I'll update it with credit.