Added my test cases, not sure if they're right though