Tweaked diamond mining code