PCLoader size

Hi together,
I tried to use your architecture for a task in my field in computational medicine, and stage1 worked fine so far, however I was wondering why the PCLoader in dataloader.py is fixed to the first five elements (dataloader/pc_loader.py:30) ?

`
 self.pc_paths = self.pc_paths[:5] 
`

would be helpful if a comment was left there with a justification.

Kind regards