Adding tests for the second homework
shorter error messages in the tests