Add tests and code to pass for each_with_object