Added functionality to collection view