Add code to pass ex1 method