Add functionality to assert