Cleanup screenshots code