Add ARC compatibility, close #4