Fixing coverage script (WIP)