Error handling and good looking view

This latest commit adds error handling and fixes the add media view