Added and slightly fixed error message screen