Started writing some test cases