Added collection input
Fixed collection input