fixing code not finished