fixing code 2