Write code to passed test