Add Error class
Minor change