Add some assertion code