Error handling, bug fixes.