The code with test