Add tests and code to pass tests