[#22] Added test cases