started test cases. Needs some more work thou