Add more error handling & pretty colors