Add error logging