Try two in fixing my own code