Added collection view and model bits