Added nicer messages to assert statements in tests