Feature/implement test cases

Just for demo case.