Writing a custom collate script