Implemented and passed a few test cases