Fixed Testing Code Going to Submit this as a base case, Will add more
cases before submitting