Minor UI update

Re-introduce .error and .success classes.