Fixing some issues with custom collection view layout.
Adding a done button to Library view controller so you can dismiss it and fixing it so it works with selected inputs.