Updated code with a dummy test