highlighted code fix final??