Basic functionality done with the given test cases handled