Implement functionality to pass tests