Add error log
Add better error handling