Add colon to hareesh's code and modified the test case