Improved the testing suite with colors, and a try-except statement