Added icon, fixed the error dialog