Edited to work with ARC

Same thing as original, but for ARC enabled apps.