[TASK] Added collection functionality