Adding a collection view delegate. Not sure if this is the approach I want, but meh.