adding test cases kinda works now