Add few more tests. Coverage - 55.8%