Tidier code