write code to pass tests