Write test for the code
Write test for the code