fixing code..