Getting Closer, got the collection view working properly