Improve ErrorMessage class