Couldn't you just make it so Grease Lightning just upgrades like Stone 1 > Stone 2 does? That way you can get GL 3 on your first Snap Punch, Demolish or Rock Breaker.