added channel dimension setup