Resolved #9 with a test, but with no UI
Added another flash message for errors
Added UI for #9
Resolve #10