Made code ARC-compatible