Adding error message

fix #8
Fix UI tests